tensor([[0.5400, 0.5400, 0.5400, 0.5400, 0.5400],
        [0.4600, 0.4600, 0.4600, 0.4600, 0.4600]], grad_fn=<RoundBackward1>)